from typing import List
from src.models.database import DatabaseConfig
from src.models.table import Station
from src.models.api_model import StationConfig


# 定义三线颜色
DEFAULT_COLOR, SF_COLOR, JJ_COLOR, BZ_COLOR = "000000", "189FA7", "0070C0", "FF0000"

# 数据库参数
DATABASE_DEV = DatabaseConfig(
    url="127.0.0.1",
    port=6041,
    user="root",
    password="Deepseek666",
    database="water",
)

STATIONS: List[Station] = [
    Station(stcd=62904500, sfsw=11.5, jjsw=13.2, bzsw=15.84, name="无为大堤"),
    Station(stcd=60115400, sfsw=9.4, jjsw=11.2, bzsw=13.4, name="城北圩"),
    Station(stcd=62900700, sfsw=8.7, jjsw=10.7, bzsw=12.7, name="江北（沈巷）长江堤"),
    Station(stcd=62906500, sfsw=10.1, jjsw=12.1, bzsw=14.1, name="万春圈堤"),
    Station(stcd=62900700, sfsw=9.4, jjsw=11.2, bzsw=12.3, name="裕溪口江堤"),
    Station(stcd=62900600, sfsw=9.5, jjsw=10.5, bzsw=12.0, name="裕溪河堤"),
    Station(stcd=62905100, sfsw=8.5, jjsw=9.5, bzsw=11.5, name="牛屯河堤"),
    Station(stcd=62904500, sfsw=11.5, jjsw=13.2, bzsw=14.5, name="惠生连圩堤"),
    Station(stcd=62904500, sfsw=11.5, jjsw=13.2, bzsw=14.5, name="永定大圩堤"),
    Station(stcd=62904500, sfsw=11.0, jjsw=13.0, bzsw=13.5, name="黑沙洲、天然洲圩"),
]

RAW_STATIONS: List[StationConfig] = [
    StationConfig(stcd=60115400, name="芜湖"),
    StationConfig(stcd=62904500, name="凤凰颈闸下"),
    StationConfig(stcd=62900700, name="裕溪闸下"),
    StationConfig(stcd=62900600, name="裕溪闸上"),
    StationConfig(stcd=62906500, name="清水"),
    StationConfig(stcd=62905100, name="新桥闸上"),
]
